How to loop code until it works in Python